import java.util.Arrays;

/**
 * 给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。
 * <p>
 * 最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
 * <p>
 * 你可以假设除了整数 0 之外，这个整数不会以零开头。
 * <p>
 * 示例 1:
 * <p>
 * 输入: [1,2,3]
 * 输出: [1,2,4]
 * 解释: 输入数组表示数字 123。
 * 示例 2:
 * <p>
 * 输入: [4,3,2,1]
 * 输出: [4,3,2,2]
 * 解释: 输入数组表示数字 4321。
 */
class Solution {

    public static void main(String[] args) {
        System.out.println(Arrays.toString(plusOne(new int[]{9, 9, 9})));
    }

    public static int[] plusOne(int[] digits) {
        int len = digits.length - 1;
        int end = digits[len];
        if (end != 9) {
            digits[len] = end + 1;
            return digits;
        } else {
            if (digits[0] == 9) {
                boolean allNine = true;
                for (int i = 1; i <= len; i++) {
                    if (digits[i] != 9) {
                        allNine = false;
                    }
                }
                if (allNine) {
                    int[] res = new int[digits.length + 1];
                    res[0] = 1;
                    return res;
                }
            }
            digits[len] = 0;
            for (int i = len - 1; i >= 0; i--) {
                if (digits[i] == 9) {
                    digits[i] = 0;
                } else {
                    digits[i] = digits[i] + 1;
                    break;
                }
            }
            return digits;
        }
    }
}